import numpy as np
import pickle

xsize = int(raw_input('X size?: '))
ysize = int(raw_input('Y size?: '))

print('The name given will be the output filename.')
parametertype = raw_input('emod or density?; ')

val1 = float(raw_input('Value 1?: '))
val2 = float(raw_input('Value 2?: '))

array = np.empty([xsize, ysize])

for x in range(xsize):
	for y in range(ysize):
		if y > x*2 - xsize/3:
			array[x,y] = val1
		else:
			array[x,y] = val2

print(array)

filename = parametertype + '.pkl'
output = open(filename, 'wb')
pickle.dump(array, output)
output.close()
print('Dumped array into pickle. Can be read into model2D.')
		
